package com.example.leetcode.c401_c500.c474;
/**
 * 在计算机界中，我们总是追求用有限的资源获取最大的收益。
 *
 * 现在，假设你分别支配着 m 个 0 和 n 个 1。另外，还有一个仅包含 0 和 1 字符串的数组。
 *
 * 你的任务是使用给定的 m 个 0 和 n 个 1 ，找到能拼出存在于数组中的字符串的最大数量。每个 0 和 1 至多被使用一次。
 *
 * 注意:
 *
 * 给定 0 和 1 的数量都不会超过 100。
 * 给定字符串数组的长度不会超过 600。
 * 示例 1:
 *
 * 输入: Array = {"10", "0001", "111001", "1", "0"}, m = 5, n = 3
 * 输出: 4
 *
 * 解释: 总共 4 个字符串可以通过 5 个 0 和 3 个 1 拼出，即 "10","0001","1","0" 。
 * 示例 2:
 *
 * 输入: Array = {"10", "0", "1"}, m = 1, n = 1
 * 输出: 2
 *
 * 解释: 你可以拼出 "10"，但之后就没有剩余数字了。更好的选择是拼出 "0" 和 "1" 。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/ones-and-zeroes
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */

/**
 * 一和零
 * @author zhanpengguo
 * @date 2020-09-18 8:14
 */
public class Solution {

    public int findMaxForm(String[] strs, int m, int n) {
        if (strs.length == 0) {
            return 0;
        }
        int[][] dp = new int[m + 1][n + 1];
        dp[0][0] = 0;
        for (String str : strs) {
            int[] count01 = count01(str);
            int zeros = count01[0];
            int ones = count01[1];
            for (int i = m; i >= zeros; i--) {
                for (int j = n; j >= ones; j--) {
                    dp[i][j] = Math.max(dp[i][j], dp[i - zeros][j - ones] + 1);
                }
            }
        }
        return dp[m][n];
    }

    private int[] count01(String str) {
        int[] ints = new int[2];
        for (int i = 0; i < str.length(); i++) {
            ints[str.charAt(i) - '0']++;
        }
        return ints;
    }

    private int get(String str, int num) {
        int count = 0;
        char[] chars = str.toCharArray();
        for (char c : chars) {
            if (num == Integer.parseInt(Character.toString(c))) {
                count++;
            }
        }
        return count;
    }

    private int digui(String[] strs, int m, int n, int i) {
        if (strs.length == 0) {
            return 0;
        }
        if (m <= 0 && n <= 0) {
            return 0;
        }
        String str = strs[i];
        int now = m >= get(str, 0) && n >= get(str, 1) ? 1 : 0;
        if (i == 0) {
            return now;
        }
        // 选当前
        int a1 = digui(strs, m - get(str, 0), n - get(str, 1), i - 1) + now;
        // 不选当前
        int a2 = digui(strs, m, n, i - 1);
        return Math.max(a1, a2);
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        String[] strs = {"10", "0", "1"};
//        int digui = solution.digui(strs, 1, 1, 2);
//        System.out.println(digui);
        int maxForm = solution.findMaxForm(strs, 1, 2);
        System.out.println(maxForm);
    }
}
